package sample.util;

import java.io.IOException;
import java.io.InputStream;


import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import com.jcraft.jsch.ChannelExec;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;
import com.jcraft.jsch.Session;
import sample.controller.MyController;

/**
 * 提供SSH连接
 *
 * @author wutc
 * @date 20201107
 *
 */

public class SSHUtils {

    final static String user = "root";// 用户名
    final static String pwd = "xxxxx";// 密码
    final static String ip = "192.168.1.30";// 服务器地址
    final static int port = 22;// 端口号
    final static int timeout = 60000000;//60000000

    public static void main(String[] args) throws Exception {

        //查看文件个数

        String commond = "free";

        String ls = SSHUtils.execCommandByJSch(ip, port,user, pwd,commond);

        System.out.println(ls);
    }


    private static final String ENCODING = "UTF-8";
    //private static final int timeout = 60 * 60 * 1000;
    private static final int defaultPort = 22;


    public static Session getJSchSession(String ip, int port, String user, String pwd, int timeout) throws JSchException {
        JSch jsch = new JSch();

        Session session = jsch.getSession(user, ip, port);
        session.setPassword(pwd);
        session.setConfig("StrictHostKeyChecking", "no"); // 第一次访问服务器时不用输入yes
        session.setConfig("PreferredAuthentications", "publickey,keyboard-interactive,password");
        session.setTimeout(timeout);
        //设置连接超时时间
        session.connect(Integer.parseInt(MyController.properties.getProperty("connectTimeOut")));
        return session;
    }

    public static String execCommandByJSch(String ip, String user, String pwd, String command) throws IOException, JSchException {
        return execCommandByJSch(ip, defaultPort, user, pwd, timeout, command, ENCODING);
    }

    public static String execCommandByJSch(String ip, int port, String user, String pwd, String command) throws IOException, JSchException {
        return execCommandByJSch(ip, port, user, pwd, timeout, command, ENCODING);
    }

    public static String execCommandByJSch(String ip, int port, String user, String pwd, int timeout, String command, String resultEncoding) throws IOException, JSchException {
        Session session = getJSchSession(ip, port, user, pwd, timeout);
        String result = execCommandByJSch(session, command, resultEncoding);
        session.disconnect();

        return result;
    }

    public static String execCommandByJSch(Session session, String command) throws IOException, JSchException {
        return execCommandByJSch(session, command, ENCODING);
    }

    public static String execCommandByJSch(Session session, String command, String resultEncoding) throws IOException, JSchException {
        ChannelExec channelExec = (ChannelExec) session.openChannel("exec");
        InputStream in = channelExec.getInputStream();
        channelExec.setCommand(command);
        channelExec.setErrStream(System.err);
        channelExec.connect();

        String result = IOUtils.toString(in, resultEncoding);

        channelExec.disconnect();

        return result;
    }


}
